// AiFileDetails.h : Declaration of the CAiFileDetails

#pragma once
#include "resource.h"       // main symbols

#include "AiShellExt_i.h"

#include <atlstr.h>


#if defined(_WIN32_WCE) && !defined(_CE_DCOM) && !defined(_CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA)
#error "Single-threaded COM objects are not properly supported on Windows CE platform, such as the Windows Mobile platforms that do not include full DCOM support. Define _CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA to force ATL to support creating single-thread COM object's and allow use of it's single-threaded COM object implementations. The threading model in your rgs file was set to 'Free' as that is the only threading model supported in non DCOM Windows CE platforms."
#endif

#include <ShlObj.h>
#include "FSHelper.h"

// CAiFileDetails

class ATL_NO_VTABLE CAiFileDetails :
	public CComObjectRootEx<CComSingleThreadModel>,
	public CComCoClass<CAiFileDetails, &CLSID_AiFileDetails>,
	public IDispatchImpl<IAiFileDetails, &IID_IAiFileDetails, &LIBID_AiShellExtLib, /*wMajor =*/ 1, /*wMinor =*/ 0>,
    public IDispatchImpl<IColumnProvider>
{
public:
	CAiFileDetails()
    {
        CFSHelper::GetFSHelper().Init();
	}
    ~CAiFileDetails()
    {
        CFSHelper::GetFSHelper().UnInit();
    }

DECLARE_REGISTRY_RESOURCEID(IDR_AIFILEDETAILS)

DECLARE_NOT_AGGREGATABLE(CAiFileDetails)

BEGIN_COM_MAP(CAiFileDetails)
	COM_INTERFACE_ENTRY(IAiFileDetails)
	COM_INTERFACE_ENTRY(IDispatch)
    COM_INTERFACE_ENTRY(IColumnProvider)
END_COM_MAP()



	DECLARE_PROTECT_FINAL_CONSTRUCT()

	HRESULT FinalConstruct()
	{
		return S_OK;
	}

	void FinalRelease()
	{
	}

public:
    // IColumnProvider methods
    STDMETHOD (Initialize)(LPCSHCOLUMNINIT psci);
    STDMETHOD (GetColumnInfo)(DWORD dwIndex, __out SHCOLUMNINFO *psci);
    STDMETHOD (GetItemData)(LPCSHCOLUMNID pscid, LPCSHCOLUMNDATA pscd, __out VARIANT *pvarData);

protected:
    STDMETHOD (GetFolderSizeColumnInfo)(DWORD dwIndex, __out SHCOLUMNINFO *psci);
    STDMETHOD (GetFolderMemoColumnInfo)(DWORD dwIndex, __out SHCOLUMNINFO *psci);

    CString    GetFolderSize(DWORD dwFileAttr, LPCTSTR szFolderPath);
    CString    GetFolderMemo(DWORD dwFileAttr, LPCTSTR szFolderPath);

protected:
    CString     m_strInitFolder;
};

OBJECT_ENTRY_AUTO(__uuidof(AiFileDetails), CAiFileDetails)
